package com.zte.ychrm.util;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Map;
import java.util.regex.Pattern;

import com.mongodb.*;
import org.bson.Document;
import org.bson.conversions.Bson;
import org.bson.types.ObjectId;

import com.mongodb.MongoClientOptions.Builder;
import com.mongodb.client.MongoCollection;
import com.mongodb.client.MongoCursor;
import com.mongodb.client.MongoDatabase;
import com.mongodb.client.MongoIterable;
import com.mongodb.client.model.Filters;
import com.mongodb.client.result.DeleteResult;

/**

    <!--mongodb java 客户端 -->
		<dependency>
			<groupId>org.mongodb</groupId>
			<artifactId>mongo-java-driver</artifactId>
			<version>3.4.1</version>
		</dependency>

 * MongoDB工具类 Mongo实例代表了一个数据库连接池，即使在多线程的环境中，一个Mongo实例对我们来说已经足够了<br>
 * 注意Mongo已经实现了连接池，并且是线程安全的。 <br>
 * 设计为单例模式， 因 MongoDB的Java驱动是线程安全的，对于一般的应用，只要一个Mongo实例即可，<br>
 * Mongo有个内置的连接池（默认为10个） 对于有大量写和读的环境中，为了确保在一个Session中使用同一个DB时，<br>
 * DB和DBCollection是绝对线程安全的<br>
 * 
 * @author http://www.cnblogs.com/zhoulf/p/4571647.html
 * @date 2015-5-29 上午11:49:49
 * @version 0.0.0
 * @Copyright (c)1997-2015 NavInfo Co.Ltd. All Rights Reserved.
 */
public enum MongoDBUtil {

    /**
     * 定义一个枚举的元素，它代表此类的一个实例
     */
    instance;

    private MongoClient mongoClient;

    static {
        System.out.println("===============MongoDBUtil初始化========================");
        // 从配置文件中获取属性值
        String ip = ConfigUtil.getPropertie("mongo.host");
        int port = Integer.parseInt(ConfigUtil.getPropertie("mongo.port"));


        //权限验证
        MongoCredential credential = MongoCredential.createScramSha1Credential(
                "ychr", "admin", "ychr".toCharArray());

        //地址
        ServerAddress addr = new ServerAddress(ip, port);




        // or, to connect to a replica set, with auto-discovery of the primary, supply a seed list of members
        // List<ServerAddress> listHost = Arrays.asList(new ServerAddress("localhost", 27017),new ServerAddress("localhost", 27018));
        // instance.mongoClient = new MongoClient(listHost);

        // 大部分用户使用mongodb都在安全内网下，但如果将mongodb设为安全验证模式，就需要在客户端提供用户名和密码：
        // boolean auth = db.authenticate(myUserName, myPassword);
        Builder options = new MongoClientOptions.Builder();
        // options.autoConnectRetry(true);// 自动重连true
        // options.maxAutoConnectRetryTime(10); // the maximum auto connect retry time
        options.connectionsPerHost(300);// 连接池设置为300个连接,默认为100
        options.connectTimeout(15000);// 连接超时，推荐>3000毫秒
        options.maxWaitTime(5000); //
        options.socketTimeout(0);// 套接字超时时间，0无限制
        options.threadsAllowedToBlockForConnectionMultiplier(5000);// 线程队列数，如果连接线程排满了队列就会抛出“Out of semaphores to get db”错误。
        options.writeConcern(WriteConcern.SAFE);//
        MongoClientOptions co = options.build();

        instance.mongoClient = new MongoClient(addr, Arrays.asList(credential), co);
//        MongoCollection<Document> resumeCollection = instance.mongoClient.getDatabase(Const.MONGODB_NAME).getCollection(Const.MONGODB_RESUME_TABLE);
//        BasicDBObject obj = new BasicDBObject();
//        obj.put("ApplicantResume.WorkExperience.JobDuty", "text");
//        //obj.put("ApplicantResume.WorkExperience.JobTitle", "text");
//        //resumeCollection.dropIndex(obj);
//        resumeCollection.createIndex(obj);

        System.out.println("===============MongoDBUtil 完成========================");
    }

    // ------------------------------------共用方法---------------------------------------------------
    /**
     * 获取DB实例 - 指定DB
     * 
     * @param dbName
     * @return
     */
    public MongoDatabase getDB(String dbName) {
        if (dbName != null && !"".equals(dbName)) {
            MongoDatabase database = mongoClient.getDatabase(dbName);
            return database;
        }
        return null;
    }

    /**
     * 获取collection对象 - 指定Collection
     * 
     * @param collName
     * @return
     */
    public MongoCollection<Document> getCollection(String dbName, String collName) {
        if (null == collName || "".equals(collName)) {
            return null;
        }
        if (null == dbName || "".equals(dbName)) {
            return null;
        }
        MongoCollection<Document> collection = mongoClient.getDatabase(dbName).getCollection(collName);
        return collection;
    }

    /**
     * 查询DB下的所有表名
     */
    public List<String> getAllCollections(String dbName) {
        MongoIterable<String> colls = getDB(dbName).listCollectionNames();
        List<String> _list = new ArrayList<String>();
        for (String s : colls) {
            _list.add(s);
        }
        return _list;
    }

    /**
     * 获取所有数据库名称列表
     * 
     * @return
     */
    public MongoIterable<String> getAllDBNames() {
        MongoIterable<String> s = mongoClient.listDatabaseNames();
        return s;
    }

    /**
     * 删除一个数据库
     */
    public void dropDB(String dbName) {
        getDB(dbName).drop();
    }

    /**
     * 查找对象 - 根据主键_id
     * 
     * @param collection
     * @param id
     * @return
     */
    public Document findById(MongoCollection<Document> coll, String id) {
        ObjectId _idobj = null;
        try {
            _idobj = new ObjectId(id);
        } catch (Exception e) {
            return null;
        }
        Document myDoc = coll.find(Filters.eq("_id", _idobj)).first();
        return myDoc;
    }

    /** 统计数 */
    public int getCount(MongoCollection<Document> coll) {
        int count = (int) coll.count();
        return count;
    }

    /** 统计数 */
    public static long getCountWithQuery(MongoCollection<Document> coll, Bson que) {
        long count = coll.count(que);
        return count;
    }


    /** 条件查询 */
    public MongoCursor<Document> find(MongoCollection<Document> coll, Bson filter) {
        return coll.find(filter).iterator();
    }
    
    /** 查询单条记录 */
    public Document findOne(MongoCollection<Document> coll, Bson filter) {
        return coll.find(filter).first();
    }

    /** 分页查询 */
    public MongoCursor<Document> findByPage(MongoCollection<Document> coll, Bson filter, int pageNo, int pageSize) {
        Bson orderBy = new BasicDBObject("_id", 1);
        return coll.find(filter).sort(orderBy).skip((pageNo - 1) * pageSize).limit(pageSize).iterator();
    }

    /** 分页查询,按条件排序 */
    public MongoCursor<Document> findByPageWithOrder(MongoCollection<Document> coll, Bson filter, Bson orderBy,int pageNo, int pageSize) {
        return coll.find(filter).sort(orderBy).skip((pageNo - 1) * pageSize).limit(pageSize).iterator();
    }

    /** 分页查询,按条件排序 */
    public MongoCursor<Document> findByPageStartWithOrder(MongoCollection<Document> coll, Bson filter, Bson orderBy,int startIdx, int pageSize) {
        return coll.find(filter).sort(orderBy).skip(startIdx).limit(pageSize).iterator();
    }
    
    /** 分页查询,不排序 */
    public MongoCursor<Document> findByPageStartWithoutOrder(MongoCollection<Document> coll, Bson filter,int startIdx, int pageSize) {
        return coll.find(filter).skip(startIdx).limit(pageSize).iterator();
    }

    /**
     * 通过ID删除
     * 
     * @param coll
     * @param id
     * @return
     */
    public int deleteById(MongoCollection<Document> coll, String id) {
        int count = 0;
        ObjectId _id = null;
        try {
            _id = new ObjectId(id);
        } catch (Exception e) {
            return 0;
        }
        Bson filter = Filters.eq("_id", _id);
        DeleteResult deleteResult = coll.deleteOne(filter);
        count = (int) deleteResult.getDeletedCount();
        return count;
    }

    /**
     * FIXME
     * 
     * @param coll
     * @param id
     * @param newdoc
     * @return
     */
    public Document updateById(MongoCollection<Document> coll, String id, Document newdoc) {
        ObjectId _idobj = null;
        try {
            _idobj = new ObjectId(id);
        } catch (Exception e) {
            return null;
        }
        Bson filter = Filters.eq("_id", _idobj);
        //coll.replaceOne(filter, newdoc); // 完全替代
        //coll.updateOne(filter, new Document("$set", newdoc));
        coll.updateOne(filter, newdoc);
        return newdoc;
    }

    public void dropCollection(String dbName, String collName) {
        getDB(dbName).getCollection(collName).drop();
    }
    
    /**
     * @date 2017年1月22日
     * @author qijun
     * @param coll
     * @param documents
     * @return
     */
    public static MongoIterable<Document> Aggregate(MongoCollection<Document> coll, List<Document> documents) {
    	return coll.aggregate(documents);
	}

    
    public static MongoIterable<Document> MapReduce(MongoCollection<Document> coll, String mapFunction,String reduceFunction){
    	return coll.mapReduce(mapFunction, reduceFunction);
    }
    
    public static MongoIterable<Document> MapReduce(MongoCollection<Document> coll, String mapFunction,String reduceFunction,Document document){
    	return coll.mapReduce(mapFunction, reduceFunction).filter(document);
    }


    //生成检索collection给定的field内容包含keyword的查询条件
    public static BasicDBObject genQueryBsonByRegexs(String[] fields, String keyword)
    {
        BasicDBObject query = new BasicDBObject();
        ArrayList<Document> queryList = new ArrayList<Document>();

        if(fields != null && fields.length > 0)
        {
            for(String field : fields)
            {
                Pattern pattern = Pattern.compile("^.*" + keyword + ".*$", Pattern.CASE_INSENSITIVE);
                BasicDBObject like = new BasicDBObject("$regex", pattern);

                Document tmp = new Document(field, like);
                queryList.add(tmp);
            }

            if(queryList.size() > 0)
            {
                query.put("$or", queryList);
            }
        }
        return query;
    }
    /**
     * 关闭Mongodb
     */
    public void close() {
        if (mongoClient != null) {
            mongoClient.close();
            mongoClient = null;
        }
    }
    
}